import pickle
import numpy
from ipywidgets import widgets

import plotly.graph_objects as go
from plotly.subplots import make_subplots

load_states_dict = open('pickled_data/general_data/states.pickle','rb')
states = pickle.load(load_states_dict)
load_states_dict.close()

textbox = widgets.Dropdown(
    description = 'State:',
    options = list(states.keys())
)

display(textbox)

load_figure = widgets.Output()

def widget_fig(state_value):
    
    state_value = textbox.value
    
    load_graph = open(f'pickled_data/graphs_pickled/{state_value}_graph.pickle','rb')
    fig = pickle.load(load_graph)
    load_graph.close()

    fig.show()
    

textbox.observe(widget_fig, names = 'value')